import pandas as pd
import random


# 定义生成随机数据的函数
def generate_random_data(num_rows):
    data = []
    for _ in range(num_rows):
        product_lv1 = random.choice(['产互KPI'])
        product_lv2 = random.choice(['业绩指标', '专业指标'])
        product_lv3 = random.choice(['AAA', 'BBB', 'CCC', 'DDD', 'EEE'])
        year = random.randint(2020, 2024)  # 生成2020到2024的随机年份
        month = random.randint(1, 12)  # 生成1到12的随机月份
        province = random.choice(['北京', '上海', '广东', '浙江', '江苏', '四川', '湖南', '湖北', '山东', '河南'])  # 随机选择省份
        city = random.choice(['北京市', '上海市', '广州市', '深圳市', '杭州市', '南京市', '成都市', '长沙市', '武汉市', '济南市', '郑州市'])  # 随机选择城市
        income = random.randint(0, 100000)  # 生成0到100000的随机收入
        budget = random.randint(100, 110000)  # 生成0到100的随机收入
        data.append([product_lv1, product_lv2, product_lv3, year, month, province, city, income, budget])
    return data


# 生成数据
num_rows = 1000  # 你可以根据需要生成更多或更少的行
data = generate_random_data(num_rows)

# 创建DataFrame
columns = ['product_lv1', 'product_lv2', 'product_lv3', 'year', 'month', 'province', 'city', 'income', 'budget']
df = pd.DataFrame(data, columns=columns)

# 显示DataFrame
print(df)

# 保存到CSV文件
df.to_csv('kpi_data.csv', index=False)